Conditional creation of a new column using another column value in Python